fix any bugs