Changed to using native sets, available in python 2.7
Changed drawing function to pass the universe rather than just cells.
Getting screen width/height on a per-use basis.
Tidying up imports.