Added small changes to dot creation