updates to learning machine setup