Added HUD functionality

Also had some comments added