Add logging; Fix key bug