Change data of key
Change data structure
Remove unnecessary code
Change data structure
Fix test cases to pass it