Added the Widget Builder class