Created tests - Fix #4

tests created