Source code of HomeWork for unit 2