trying make code work by removing line 5