add source code for HW2 after execution