complete Assignment 2