This can still be improved due to the messup at the end, will come back to this later.