Tryign to fix code problem