Changed header file
Added drawing back in