Added support for categories
Added support for categories

s-ramaswamy/CS-229-project,s-ramaswamy/CS-229-project,s-ramaswamy/CS-229-project,s-ramaswamy/CS-229-project,s-ramaswamy/CS-229-project,s-ramaswamy/CS-229-project