Added official solution for ex2
Removed unused import