Simplify and just use input element