started writing trainParams

updated TODO and the psuedo section in main