Added a test class :)