Clean up code
Add tags