Update GraphicsUnit.scala

Refactoring