Give class instead of using div