Starting to add some logic
Updated readme