Add check_play_again function