- [x] Take feedback from GJ/Revanth and come up with changes in the functionalities - [x] Create high-fidelity mocks for all the functionalities