Add entity generation function to the tool but unfinished