Issue/3

fixes #3