Working on add widget functionality (#10).