Add model scoring code