Adds tests for logging and custom logging, moves logging flag test to executing code vs callback

This will complete #206