Add little helper class