Improved some stuff and added pretty HUD elements