Add loading animation (not fully implemented)
add loading state