edit the code , looks nicer now