Update Ex1. Optimize classes