Main.r

functional version but using correct data for training / validation
needs to be implemented