Added categoryCell and better ui