updating code a bit