Adding Train/Validate/Test sets, plotting validation accuracy