Improved code, added computer AI, added loop to play again