Added solution and updated readme