Add in more props