Issue 5

fixes #5