Minor Fix
Changed some things