added tests for user story 2