Partially fixed the main button