Adding category to New and Edit forms