Added tests

Added tests and "SetUp" method