Added logic to get class schedule