minor fixes

fixed the main button