Split JS into separate file and added some more stylings