[TEST] Adding tests