Updated unit edit