Adding functionality - working on data processing