refactor
added the idea