Updates to widget to allow for multi level structure