Added new pictures and re-formatted code