Fixes #116
We don't need to separate widget/onscreen from the main module, let's merge them together.