styling and both branch functions working