More changes, tried to get branch in the prompt but it didn't work