My attempts at the Python challenge