Add simple test
Mocking algorithm
add TODO