Added widget functionality