homework-5 (first commit)

Not fully working.