New updated code (version 1.1)