changed some code
removed print statemet
should be on new branch
should be on new branch 2