fixes final bugs