Issue 6

fixed #6