Update my assignment2