Updated with a test