Add plotting loop