Fixes #3

Code