Added logic class