small style improvements in extension