Updates to unit script, hopefully looks better.