Suggestion to refactor code into simple way -React