working correctly but still won't work correctly for testing