Add missing CSS class to button
Fix pep8 formatting errors