Better python CLI integration