Working code + setup file + tests that pass