minor refactor
removed print
minor refactor