Finished hw1, haven't tested yet