Added tests

Created tests.