Added solution and drivers code