Tests created - Fixes #4

tests created