Add initial ball movement code