small program on python