Finished the last section of the homework, ready for submission now!