Adding python code for UI, still in test 1.