Add two hooks
Use session storage to remember tab