Issue/2

fixes #2