Add little fixes