Added a drwaing function. Not perfect