Tests added

Hello. I think some tests wont hurt :)