removed unnecesary imports + print statements