Unfinished linear assignment

Still working on the dimensional issue between test and train