Simple program using Python