Assignment 01 solution