Add mockery, close #5