https://scholarshipshome.info/sta301-assignment-1-solution/