remove print statements and fix indentation