Code works for 1