Button animation swift 3