Finally working through python