Update code examples

Better showing them in python shell style