improve e302 fix code

Signed-off-by: Thomas Hisch <t.hisch@gmail.com>
include spaces in the 'class' or 'def' checks
Merge pull request #166 from thisch/improve_e302

improve e302 fix code