Latest homework submission with everything working