Assignment 2, prog 1