changed to code to better match python styleguide