started adding test code