First attempt at Python module