Updating STAN model to use STAN's syntax better