Closes issue #5

fixed test