Cleaned up model selection and validation notebook