WIP
Remove print statement
Make ToolRegistry capable of distinguishing between different widget classes