forgot to add new files - fixed issue with tool display