more code

edited