Assignment 5.1 version 01