The latest work. Should support nb classification with the required parameters