Just a sketch of unit tests.
Removing unnecessary print.
Some todo's.