Bug fix+TODO