Updated codes and added images

Some code has been updated and screen shots of the layouts plus the UML
diagram has been added in an images folder