Fixed the problem with drawing units code

Fixed the issue identified by daniel. However the project can not be
compiled due to errors with shotHit function on the shooting code