Fix pylint issues

Signed-off-by: Stephen Finucane <stephenfinucane@hotmail.com>