Commit fixing bug 4

Fixes #4