add so much more python code yeaaaaa