Add snippets for class and main