Addition of Assignment 2 material