Closes #5

Fixed it!