Add game over logic and fix some balancing