Finally got my code to work :D