Added swap size input

- Change the swap size unit to GB from MB.
- Added swap size as an optional input.
- Fixed readme regarding the swap size.
- Fixed tests.