HW6 basic task completed. To be continued