Add partially finished python code