Changed widget layout to flex