Cleaned up main(), added play again checker