tests

Hi! I've added tests for you.