First small python project