My first attempt at making a python program