Day 4

Fixed training set