updated index file and wrote models to pass test