Updated Logic Class

Added some basic logic.