Better loading process for .py