homework-4 (first commit)

Not working version.