Update README.md

README.md

@@ -63,9 +63,9 @@ pip install -r [requirements.txt](http://_vscodecontentref_/4)
 
 ### Example Use
 
-Task 1: Molecule Sequence Generation
+Task 1: Molecule Sequence Generation \
 Generate new pseudo-Natural Product SMILES sequences using the pretrained model:
-```
+```python
 from mol_generation import run_generation
 
 # Configure and run generation
@@ -83,7 +83,7 @@ Output: CSV file with columns [Molecule, Log-Likelihood]
 
 Alternatively, use the CLI:
 
-```
+```bash
 python3 main.py \
   --task generate \
   --model_names rozariwang/M1-AIS-rds \