Can now handle a main function with multiple basic blocks