Fix any bugs