Add more functionality to .small