First attempt at python (incomplete)