Initial code for summer internship