Added test folder and fixed whitespace according to PEP8