edited code a bit for test testing