sedikit perubahan dan perapihan source code