Added edit category functionality - some minor updates required