Final project for SDEV1 - using Prof smith's code