Starting to work on implementing additional models