made changes to my solution