fixes #91
updating code