Small updates to theme