Transient version correction according to python style guidelines and functionality

All imports in the beginning of script

Defined dimensionality of coords N_depths = len(depths) N_lat = len(lat) N_lon = len(lon)