first draft of simple_atmos.py. got something working