Added stuff for taag model