Moved tests from the ViewController to the tests class (still have to cleanup)