Still trying to fix game logic :|