My updated code, added bootstrap and added layout