code cleaned up after pylint evaluation