Store original code from HW1