Issues/4

fixed issue #4.