Some changes to the classes - comments and adding a test class