first try, added code