Fixed layout issues

Fixed #11 
This should also toggle the state of the `persist` variable :)