TEST: Removed all the prints from test_optimize