Assignment 6 for IMPR