minor changes made on draw method