change based on mentor's suggestion