Adding Category to save method