Hotfix in screens.py: __init__ -> main
__init__ method cannot return anything, so I've changed it back to
main() as temporary solution.