Fixed issues with branching and control flow instructions