add lib for better models