My first attempt at learning python