making traj adhere to pep 8