PEP-8 comments

Signed-off-by: Michal Čihař <michal@cihar.com>
Simplify condition

Signed-off-by: Michal Čihař <michal@cihar.com>