Adding correct styling