Just for HW5. To be delted