Updates to styling and basic functionality