Issue/4

fixed issue #4