Updated to use PEP8 styling