forgot to add newly public class