Refactor code
Merge pull request #4 from BunanaKing/fix/kill_balls

Refactor code