diff --git a/app/lib/agents/values_agent.dart b/app/lib/agents/values_agent.dart
index 26b2dea..cf5a21d 100644
--- a/app/lib/agents/values_agent.dart
+++ b/app/lib/agents/values_agent.dart
@@ -6,6 +6,7 @@ import 'package:app/models/chat_message.dart';
 import 'package:app/providers/prompts_providers.dart';
 import 'package:app/services/ai_service.dart';
 import 'package:flutter_riverpod/flutter_riverpod.dart';
+import 'dart:developer' as logger;
 
 final attributesActionsProvider = FutureProvider<List<AttributesActionObj>>((ref) async {
   final asyncSummaryPrompt = ref.watch(summaryPromptProvider);
@@ -69,19 +70,19 @@ class AttributesAgent {
     try {
       return await analyze(summaryPrompt);
     } catch (err) {
-      print('Error while analyzing the summary');
-      print(err);
+      logger.log('Error while analyzing the summary');
+      logger.log(err.toString());
       return [];
     }
   }
 
   Future<List<AttributesActionObj>> analyze(String summaryPrompt) async {
     final chatWithPrompts = _createChatWithPrompts(summaryPrompt);
-    print(chatWithPrompts[0].content);
-    print(chatWithPrompts[1].content);
-    print(chatWithPrompts[2].content);
+    logger.log(chatWithPrompts[0].content);
+    logger.log(chatWithPrompts[1].content);
+    logger.log(chatWithPrompts[2].content);
     final aiResponse = await aiService.respondToChat(chatWithPrompts);
-    print(aiResponse);
+    logger.log(aiResponse.toString());
     return _parseAttributeActionsFromAIResponse(aiResponse.content);
   }
 
@@ -127,7 +128,7 @@ class AttributesAgent {
     try {
       return AttributesActionObj.fromJson(json.decode(jsonString));
     } catch (err) {
-      print(err);
+      logger.log(err.toString());
       // TODO: how to error handle here?
       return null;
     }
diff --git a/app/lib/common/chat_widget.dart b/app/lib/common/chat_widget.dart
index 9c5d51c..aa7a707 100644
--- a/app/lib/common/chat_widget.dart
+++ b/app/lib/common/chat_widget.dart
@@ -19,7 +19,7 @@ class ChatWidget extends StatelessWidget {
     required UserObj user,
     PersonalityObj? assistantPersonality,
     required this.onMessageAdded,
-    required List<AsyncValue<ChatMessageObj>> messages,
+    required List<types.Message> messages,
     super.key,
   }) {
     this.user = types.User(
@@ -27,7 +27,7 @@ class ChatWidget extends StatelessWidget {
       firstName: 'You',
     );
 
-    this.assistantUser = assistantPersonality != null
+    assistantUser = assistantPersonality != null
       ? types.User(
         id: assistantPersonality.id!,
         firstName: assistantPersonality.name,
@@ -38,7 +38,7 @@ class ChatWidget extends StatelessWidget {
       );
 
     // TODO: Think about this performace... every time we add a new Chat message the entire widget is reloaded and all the messages are reconverted
-    this.messages = messages
+    messages = messages
       .map(_toTextMessage)
       .whereType<types.Message>()
       .toList();
diff --git a/app/lib/common/utils.dart b/app/lib/common/utils.dart
index 2438f16..d86d407 100644
--- a/app/lib/common/utils.dart
+++ b/app/lib/common/utils.dart
@@ -1,5 +1,5 @@
-import 'package:tuple/tuple.dart';
 import 'package:flutter_riverpod/flutter_riverpod.dart';
+import 'package:tuple/tuple.dart';
 import 'package:uuid/uuid.dart';
 
 var _uuid = const Uuid();
diff --git a/app/lib/mocks/data/firebase_test_data.dart b/app/lib/mocks/data/firebase_test_data.dart
index bf973a2..0a395bb 100644
--- a/app/lib/mocks/data/firebase_test_data.dart
+++ b/app/lib/mocks/data/firebase_test_data.dart
@@ -4,7 +4,7 @@ import 'package:app/models/journal_entry.dart';
 import 'package:app/models/summary.dart';
 import 'package:app/models/user.dart';
 import 'package:app/models/attribute.dart';
-import 'package:fake_cloud_firestore/fake_cloud_firestore.dart';
+
 
 const testUserId = 'id_test_user';
 const testUser = UserObj(id: testUserId, name: 'John Doe');
diff --git a/app/lib/repositories/attribute_repository.dart b/app/lib/repositories/attribute_repository.dart
index f39ab49..407d03c 100644
--- a/app/lib/repositories/attribute_repository.dart
+++ b/app/lib/repositories/attribute_repository.dart
@@ -4,6 +4,7 @@ import 'package:app/providers/encrypter_provider.dart';
 import 'package:app/providers/firebase_providers.dart';
 import 'package:cloud_firestore/cloud_firestore.dart';
 import 'package:flutter_riverpod/flutter_riverpod.dart';
+import 'dart:developer' as logger;
 
 final attributeRepositoryProvider = Provider<BaseAttributeRepository>((Ref ref) {
   final firestore = ref.read(firebaseFirestoreProvider);