added categories. closed issue #4