Minor bug fixes, adding screenshots