need to add in noise classes