Class_5 updated code