diff --git a/README.md b/README.md
index 5a711c6..f3d6a7a 100644
--- a/README.md
+++ b/README.md
@@ -5,5 +5,5 @@ This project is based on the following resources:
 
 ## Use
 ```bash
-python main.py --dataset folder --dataroot data/maps --imageSize 256
+python main.py
 ```
diff --git a/main.py b/main.py
index f388b39..5b239ea 100644
--- a/main.py
+++ b/main.py
@@ -20,11 +20,11 @@ import models.mlp as mlp
 if __name__=="__main__":
 
     parser = argparse.ArgumentParser()
-    parser.add_argument('--dataset', required=True, help='cifar10 | lsun | imagenet | folder | lfw ')
-    parser.add_argument('--dataroot', required=True, help='path to dataset')
+    parser.add_argument('--dataset', default='folder', help='cifar10 | lsun | imagenet | folder | lfw ')
+    parser.add_argument('--dataroot', default='data/maps', help='path to dataset')
     parser.add_argument('--workers', type=int, help='number of data loading workers', default=2)
     parser.add_argument('--batchSize', type=int, default=64, help='input batch size')
-    parser.add_argument('--imageSize', type=int, default=64, help='the height / width of the input image to network')
+    parser.add_argument('--imageSize', type=int, default=256, help='the height / width of the input image to network')
     parser.add_argument('--nc', type=int, default=3, help='input image channels')
     parser.add_argument('--nz', type=int, default=100, help='size of the latent z vector')
     parser.add_argument('--ngf', type=int, default=64)
@@ -41,8 +41,6 @@ if __name__=="__main__":
     parser.add_argument('--clamp_upper', type=float, default=0.01)
     parser.add_argument('--Diters', type=int, default=5, help='number of D iters per each G iter')
     parser.add_argument('--noBN', action='store_true', help='use batchnorm or not (only for DCGAN)')
-    parser.add_argument('--mlp_G', action='store_true', help='use MLP for G')
-    parser.add_argument('--mlp_D', action='store_true', help='use MLP for D')
     parser.add_argument('--n_extra_layers', type=int, default=0, help='Number of extra layers on gen and disc')
     parser.add_argument('--experiment', default=None, help='Where to store samples and models')
     parser.add_argument('--adam', action='store_true', help='Whether to use adam (default is rmsprop)')