Nit: This should be indented an additional two spaces. We use four-space indents in Python.