Minor design tweaks and update readme