sampling tests
models + sampling tests
typo
fixed argument errors
trace tests
 added step tests
forgot imports
changes for step methods
test fixes