Small modifications using div