Add training and validation. Add printing functions