Add working quiz logic