write code

keep calm and write code; python