Updated with assignment 2, and updated styling.