Improving Code strucutre
Added pictures