Removed unnecessary modules and print statements