Teaching myself Python by writing a program!