added assignment 4.1