Fixed error when no category is given
Fixed display bug