cleaned code and added pictures