Logging

Moves print statements to python logger
Lots of documentation improvements