Added some tests, got play again loop working