Improve setup and utility methods
Add label feature probability analysis