Added solution for diamond