Fix pylint complaints

Based on #106.