Made the save an load entity buttons work