Updated Assignment 5 with comments