A simple code to learn Python