PEP8
Add set_level function