Support hiding input

Changed expect structure from a list to a dict so we can have more
control over inputs.