Python attempt 1