fixed first problem