work on cs758 assignment 3