Add needed models