Making sure classes work properly