Code works for 1-6