Run cargo fmt and start creating unit tests

[paazmaya]

Here are some first unit tests that could be a starting point for adding more.

Also ran cargo fmt, hence the other code changes...