started to work on user story 4

interesting problem - solved but have to work on the types a bit