Fix pep8 on repo
Fix custom buttons and few modifications