Worked on a bug. Fixes #1.