updated code and images