cleaned up extra print statements
cleaned up extra imports