Trying out class based approach