For CS561 Homework Assignment 1