Added type key to python script