diff --git a/README.md b/README.md
index 876826d..b0b0264 100644
--- a/README.md
+++ b/README.md
@@ -92,7 +92,7 @@ After starting up, most of the screen will be empty.
 ┃                                        ┃
 ┃                                        ┃
 ┃                                        ┃
-┃ CBM 3001 MEMTEST, 2022 S3LPH.ME     ***┃
+┃CBM 3001 MEMTEST4K , 2022 S3LPH.ME   ***┃
 ┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
 ```
 
diff --git a/src/memtest6502.asm b/src/memtest6502.asm
index a658163..5289ddd 100644
--- a/src/memtest6502.asm
+++ b/src/memtest6502.asm
@@ -4,9 +4,25 @@
 #endif
 
 #ifndef _MEMEND
-#define _MEMEND $4000
+#define _MEMEND $4000	
 #endif
 
+#ifndef _MEMSIZE
+#define _MEMSIZE "??K"
+#if _MEMEND = $1000
+#define _MEMSIZE "4K "
+#endif
+#if _MEMEND = $2000
+#define _MEMSIZE "8K "
+#endif
+#if _MEMEND = $4000
+#define _MEMSIZE "16K"
+#endif
+#if _MEMEND = $8000
+#define _MEMSIZE "32K"
+#endif
+#endif
+	
 	scrptr = $8000
 	eoscr = $83e7		; last screen address
 	lastline = $83c0	; start of last line
@@ -104,7 +120,7 @@ textloop:
 	lda infotext,Y
 	sta lastline,Y
 	iny
-	cpy #$20
+	cpy #$22
 	bne textloop
 
 	;; set up first pass with $FF
@@ -486,7 +502,7 @@ passbytes:
 passchars:
 	.asc "FF00AA55XX0102040810204080FEFDFBF7EFDFBF7F"
 infotext:
-	.asc " CBM 3001 MEMTEST, 2022 S3LPH.ME"
+	.asc "CBM 3001 MEMTEST", _MEMSIZE, ", 2022 S3LPH.ME"
 
 nmiv:
 	;; NMI: Show ! in status