refine main page button -- not finished -- keep working at home