### State management in react is a f*king mess
### Update to TypeScript