Code cleanup
Implement Category parameter