polishing function.py, had to fail another 2 tests in pretty printing