Some code, still need to test

Not sure if this will work :/