add test plus little fixes