Rough version of the tool is working :)