changed input to custom class