Actual code with tests