improve style
add missing comma
fix imports
Merge pull request #1 from lucasweiblen/lw-fixstyle

Improve style