1 files changed, 8 insertions, 5 deletions

diff --git a/src/repl.c b/src/repl.c
index c63ad44..ed829d1 100644
--- a/src/repl.c
+++ b/src/repl.c
@@ -5,7 +5,6 @@ inputBuffer *newInputBuffer(void){
 	inputBuffer *in = malloc(sizeof(inputBuffer));
 	in->buffer = NULL;
 	in->inputSize = 0;
-	
 	return in;
 }
 
@@ -18,13 +17,17 @@ void getInput(inputBuffer *in){
 	size_t n;
 	ssize_t r = getline(&(in->buffer), &n, stdin);
 	if(r <= 0){
-		printf("Error\n");
+		fprintf(stderr, "Error\n");
 		exit(r);
 	}
 	in->inputSize = r-1;
 	in->buffer[in->inputSize] = 0;
 }
 
-void prompt(void){
-	printf("db > ");
-}
\ No newline at end of file
+void prompt(database *db){
+	if(db == NULL){
+		printf("db > ");
+	}else{
+		printf("%s > ", db->name);
+	}
+}