Added import model functionality