Let's make it a class