Add indication of loading state