Initial code + readme
Fix muddled if statement