Added specs to test the main classes