Add errorhandling