Make init.py slimmer