Made corrections to stan model