added print back in
replaced exit with return