Added base logic for edit category screen