Work for HW2 in software engineering