improve sample load
improve setup models