Woops! Edited the wrong branch, fixes #1

But since is just a practice exercise, I guess that's okay.