Finished assignment three. (kind of)