adding mat257 assignment 4