Skip first model and fix probability calc