cleaned up the code on the branch file a little