started adding while() test