Simplify the module down, the old way was confusing
Styling