I got rid of some unused imports and print statements