Added ball movement routine