Minor styling updates -- need to revisit