Improve readme and create models