Added imports to reduce compiler warnings

Helps address this issue: Lots of warnings in generated output #224