code for deep learning class project