Modified demo and also added input style message