Sample code on python basics