Add this in constructor