Implemented working DB, add and remove functions. Started to deal with the logic.