Bouncing balls with Apple's Swift